hackerrank transpose and flatten problem solution in python programming